import numpy as np
import torch
import random
import math
import torch.nn.functional as F
a = torch.rand((3, 1280, 1280))
print(a)
result = torch.chunk(a, 3, dim=0)  # 说明就是按照通道进行切割，返回的结果为一个元祖
print(type(result))
print(len(result))
print(result[0].shape)
print(result[1].shape)
print(result[2].shape)
